/* USER CODE BEGIN Header */
/**
  ******************************************************************************
  * @file           : main.c
  * @brief          : Main program body
  ******************************************************************************
  * @attention
  *
  * Copyright (c) 2023 STMicroelectronics.
  * All rights reserved.
  *
  * This software is licensed under terms that can be found in the LICENSE file
  * in the root directory of this software component.
  * If no LICENSE file comes with this software, it is provided AS-IS.
  *
  ******************************************************************************
  */
/* USER CODE END Header */
/* Includes ------------------------------------------------------------------*/
#include "main.h"
#include "octospi.h"
#include "spi.h"
#include "usart.h"
#include "gpio.h"

/* Private includes ----------------------------------------------------------*/
/* USER CODE BEGIN Includes */
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

#include "psram.h"
#include "norflash.h"
/* USER CODE END Includes */

/* Private typedef -----------------------------------------------------------*/
/* USER CODE BEGIN PTD */

/* USER CODE END PTD */

/* Private define ------------------------------------------------------------*/
/* USER CODE BEGIN PD */

/* USER CODE END PD */

/* Private macro -------------------------------------------------------------*/
/* USER CODE BEGIN PM */

/* USER CODE END PM */

/* Private variables ---------------------------------------------------------*/

/* USER CODE BEGIN PV */

/* USER CODE END PV */

/* Private function prototypes -----------------------------------------------*/
void SystemClock_Config(void);
void MPU_Config(void);
/* USER CODE BEGIN PFP */

/* USER CODE END PFP */

/* Private user code ---------------------------------------------------------*/
/* USER CODE BEGIN 0 */
uint8_t norflash_buf[1024];
void NORFLASH_test(void)
{
  uint8_t flash_id[3];
  uint32_t err1 = 0, err2 = 0;
  NORFLASH_DeviceID(flash_id);
  NORFLASH_Reset();
  NORFLASH_DeviceID(flash_id);
  NORFLASH_Reset();
  NORFLASH_DeviceID(flash_id);
  NORFLASH_Reset();
  printf("NORFLASH ID: 0x%x 0x%x 0x%x \r\n", flash_id[0], flash_id[1], flash_id[2]);

  NORFLASH_EraseBlock(0);
  for (volatile int i = 0; i < 64; i++)
  {
    NORFLASH_Read(norflash_buf, i * sizeof(norflash_buf), sizeof(norflash_buf));
    for (volatile int j = 0; j < 1024; j++)
    {
      if (norflash_buf[j] != 0xff)
      {
        err1++;
      }
    }
  }
  printf("nor flash read test (err = %d). \r\n", err1);
  ///////////////////////////////////////////////////////////////////////////////
  for (volatile int i = 0; i < 64; i++)
  {
    memset(norflash_buf, 0xa5, sizeof(norflash_buf));
    NORFLASH_Write(norflash_buf, i * sizeof(norflash_buf), sizeof(norflash_buf));
  }
  for (volatile int i = 0; i < 64; i++)
  {
    NORFLASH_Read(norflash_buf, i * sizeof(norflash_buf), sizeof(norflash_buf));
    for (volatile int j = 0; j < sizeof(norflash_buf); j++)
    {
      if (norflash_buf[j] != 0xa5)
      {
        err2++;
      }
    }
  }
  printf("nor flash write test (err = %d). \r\n", err2);
}

uint8_t psram_id[8];
uint8_t norflash_id[3];
int norflash_psram_check(void)
{
  uint8_t psram_check = 0;
  uint8_t norflash_check = 0;
  OSPI_NORPSRAM_ID(psram_id);
  NORFLASH_DeviceID(norflash_id);

  printf("NORPSRAM ID: 0x%x 0x%x \r\n", psram_id[0], psram_id[1]);
  printf("NORFLASH ID: 0x%x 0x%x 0x%x \r\n", norflash_id[0], norflash_id[1], norflash_id[2]);

  if (psram_id[0] == 0x0d && psram_id[1] == 0x5d)
  {
    psram_check = 1;
  }

  if (norflash_id[0] == 0xc2 && norflash_id[1] == 0x20 && norflash_id[2] == 0x16)
  {
    norflash_check = 1;
  }

  if (psram_check && norflash_check)
  {
    return 1;
  }
  else
  {
    return 0;
  }
}

#define APP_SIZE (512 * 1024)
uint8_t app_buf[NORFLASH_PAGE_SIZE];
int copy_app(void)
{
  uint8_t flag = 1;
  for (volatile int i = 0; i < (APP_SIZE / NORFLASH_PAGE_SIZE); i++)
  {
    NORFLASH_Read256B(app_buf, i * sizeof(app_buf));
    OSPI_NORPSRAM_WRITE(i * sizeof(app_buf), app_buf, sizeof(app_buf));
  }

  if (flag == 0)
  {
    flag = 1;
    NORFLASH_Read(app_buf, 0, NORFLASH_SECTOR_SIZE);
    for (volatile int i = 0; i < NORFLASH_SECTOR_SIZE; i++)
    {
      printf("0x%x ", app_buf[i]);
      if (((i+1) % 16) == 0)
      {
        printf("\r\n");
      }
    }
  }

  return 1;
}

#define APPLICATION_ADDRESS 0x90000000UL
typedef void (*appFun)(void);
appFun App;
void go_app(void)
{
    __set_PRIMASK(1);
    SysTick->CTRL = 0;
    SysTick->LOAD = 0;
    SysTick->VAL = 0;
    for (int i = 0; i < 8; i++)
    {
        NVIC->ICER[i] = 0xFFFFFFFF;
        NVIC->ICPR[i] = 0xFFFFFFFF;
    }
    __set_PRIMASK(0);
    uint32_t *vector_table = (uint32_t *)APPLICATION_ADDRESS;
    uint32_t stack_point = vector_table[0];
    uint32_t entry_point = vector_table[1];
    if ((entry_point & 0xf0000000) == APPLICATION_ADDRESS)
    {
      App = (appFun)entry_point;
      __set_MSP(stack_point);
      __set_CONTROL(0);
      App();
    }
    else
    {
      //OCTOSPI1->CR = OCTOSPI1->CR & 0x1ffffffe;
      HAL_OSPI_DeInit(&hospi1);
      HAL_SPI_DeInit(&hspi2);
    }
}
/* USER CODE END 0 */

/**
  * @brief  The application entry point.
  * @retval int
  */
int main(void)
{
  /* USER CODE BEGIN 1 */

  /* USER CODE END 1 */

  /* MPU Configuration--------------------------------------------------------*/
  //MPU_Config();

  /* MCU Configuration--------------------------------------------------------*/

  /* Reset of all peripherals, Initializes the Flash interface and the Systick. */
  HAL_Init();

  /* USER CODE BEGIN Init */

  /* USER CODE END Init */

  /* Configure the system clock */
  SystemClock_Config();

  /* USER CODE BEGIN SysInit */

  /* USER CODE END SysInit */

  /* Initialize all configured peripherals */
  MX_GPIO_Init();
  MX_USART1_UART_Init();
  MX_OCTOSPI1_Init();
  MX_SPI2_Init();
  /* USER CODE BEGIN 2 */
  printf("STM32H7B0VB psram boot. \r\n");
  OSPI_NORPSRAM_Reset(); 
  NORFLASH_Reset();
  //NORFLASH_test();
  if (norflash_psram_check())
  {
    OSPI_NORPSRAM_QSPI();
    copy_app();
    if (OSPI_NORPSRAM_MemoryMapped() != OSPI_OK)
    {
      printf("!!! OSPI_NORPSRAM_MemoryMapped() error !!! \r\n");
    }
    else
    {
      printf("!!! OSPI_NORPSRAM_MemoryMapped() ok !!! \r\n");
      go_app();
    }
  }
  /* USER CODE END 2 */

  /* Infinite loop */
  /* USER CODE BEGIN WHILE */
  while (1)
  {
    /* USER CODE END WHILE */

    /* USER CODE BEGIN 3 */
  }
  /* USER CODE END 3 */
}

/**
  * @brief System Clock Configuration
  * @retval None
  */
void SystemClock_Config(void)
{
  RCC_OscInitTypeDef RCC_OscInitStruct = {0};
  RCC_ClkInitTypeDef RCC_ClkInitStruct = {0};

  /*AXI clock gating */
  RCC->CKGAENR = 0xFFFFFFFF;

  /** Supply configuration update enable
  */
  HAL_PWREx_ConfigSupply(PWR_LDO_SUPPLY);

  /** Configure the main internal regulator output voltage
  */
  __HAL_PWR_VOLTAGESCALING_CONFIG(PWR_REGULATOR_VOLTAGE_SCALE3);

  while(!__HAL_PWR_GET_FLAG(PWR_FLAG_VOSRDY)) {}

  /** Initializes the RCC Oscillators according to the specified parameters
  * in the RCC_OscInitTypeDef structure.
  */
  RCC_OscInitStruct.OscillatorType = RCC_OSCILLATORTYPE_HSI;
  RCC_OscInitStruct.HSIState = RCC_HSI_DIV1;
  RCC_OscInitStruct.HSICalibrationValue = 64;
  RCC_OscInitStruct.PLL.PLLState = RCC_PLL_ON;
  RCC_OscInitStruct.PLL.PLLSource = RCC_PLLSOURCE_HSI;
  RCC_OscInitStruct.PLL.PLLM = 32;
  RCC_OscInitStruct.PLL.PLLN = 64;
  RCC_OscInitStruct.PLL.PLLP = 2;
  RCC_OscInitStruct.PLL.PLLQ = 2;
  RCC_OscInitStruct.PLL.PLLR = 2;
  RCC_OscInitStruct.PLL.PLLRGE = RCC_PLL1VCIRANGE_1;
  RCC_OscInitStruct.PLL.PLLVCOSEL = RCC_PLL1VCOWIDE;
  RCC_OscInitStruct.PLL.PLLFRACN = 0;
  if (HAL_RCC_OscConfig(&RCC_OscInitStruct) != HAL_OK)
  {
    Error_Handler();
  }

  /** Initializes the CPU, AHB and APB buses clocks
  */
  RCC_ClkInitStruct.ClockType = RCC_CLOCKTYPE_HCLK|RCC_CLOCKTYPE_SYSCLK
                              |RCC_CLOCKTYPE_PCLK1|RCC_CLOCKTYPE_PCLK2
                              |RCC_CLOCKTYPE_D3PCLK1|RCC_CLOCKTYPE_D1PCLK1;
  RCC_ClkInitStruct.SYSCLKSource = RCC_SYSCLKSOURCE_HSI;
  RCC_ClkInitStruct.SYSCLKDivider = RCC_SYSCLK_DIV1;
  RCC_ClkInitStruct.AHBCLKDivider = RCC_HCLK_DIV1;
  RCC_ClkInitStruct.APB3CLKDivider = RCC_APB3_DIV2;
  RCC_ClkInitStruct.APB1CLKDivider = RCC_APB1_DIV2;
  RCC_ClkInitStruct.APB2CLKDivider = RCC_APB2_DIV2;
  RCC_ClkInitStruct.APB4CLKDivider = RCC_APB4_DIV2;

  if (HAL_RCC_ClockConfig(&RCC_ClkInitStruct, FLASH_LATENCY_2) != HAL_OK)
  {
    Error_Handler();
  }
}

/* USER CODE BEGIN 4 */

/* USER CODE END 4 */

/* MPU Configuration */

void MPU_Config(void)
{
  MPU_Region_InitTypeDef MPU_InitStruct = {0};

  /* Disables the MPU */
  //HAL_MPU_Disable();

  /** Initializes and configures the Region and the memory to be protected
  */
  MPU_InitStruct.Enable = MPU_REGION_ENABLE;
  MPU_InitStruct.Number = MPU_REGION_NUMBER0;
  MPU_InitStruct.BaseAddress = 0x90000000;
  MPU_InitStruct.Size = MPU_REGION_SIZE_2MB;
  MPU_InitStruct.SubRegionDisable = 0x0;
  MPU_InitStruct.TypeExtField = MPU_TEX_LEVEL1;
  MPU_InitStruct.AccessPermission = MPU_REGION_FULL_ACCESS;
  MPU_InitStruct.DisableExec = MPU_INSTRUCTION_ACCESS_ENABLE;
  MPU_InitStruct.IsShareable = MPU_ACCESS_NOT_SHAREABLE;
  MPU_InitStruct.IsCacheable = MPU_ACCESS_CACHEABLE;
  MPU_InitStruct.IsBufferable = MPU_ACCESS_BUFFERABLE;

  HAL_MPU_ConfigRegion(&MPU_InitStruct);

  MPU_InitStruct.Enable = MPU_REGION_ENABLE;
  MPU_InitStruct.Number = MPU_REGION_NUMBER1;
  MPU_InitStruct.BaseAddress = 0x90200000;
  MPU_InitStruct.Size = MPU_REGION_SIZE_2MB;
  MPU_InitStruct.SubRegionDisable = 0x0;
  MPU_InitStruct.TypeExtField = MPU_TEX_LEVEL0;
  MPU_InitStruct.AccessPermission = MPU_REGION_FULL_ACCESS;
  MPU_InitStruct.DisableExec = MPU_INSTRUCTION_ACCESS_ENABLE;
  MPU_InitStruct.IsShareable = MPU_ACCESS_SHAREABLE;
  MPU_InitStruct.IsCacheable = MPU_ACCESS_NOT_CACHEABLE;
  MPU_InitStruct.IsBufferable = MPU_ACCESS_NOT_BUFFERABLE;

  HAL_MPU_ConfigRegion(&MPU_InitStruct);
  /* Enables the MPU */
  HAL_MPU_Enable(MPU_PRIVILEGED_DEFAULT);
}

/**
  * @brief  This function is executed in case of error occurrence.
  * @retval None
  */
void Error_Handler(void)
{
  /* USER CODE BEGIN Error_Handler_Debug */
  /* User can add his own implementation to report the HAL error return state */
  __disable_irq();
  while (1)
  {
  }
  /* USER CODE END Error_Handler_Debug */
}

#ifdef  USE_FULL_ASSERT
/**
  * @brief  Reports the name of the source file and the source line number
  *         where the assert_param error has occurred.
  * @param  file: pointer to the source file name
  * @param  line: assert_param error line source number
  * @retval None
  */
void assert_failed(uint8_t *file, uint32_t line)
{
  /* USER CODE BEGIN 6 */
  /* User can add his own implementation to report the file name and line number,
     ex: printf("Wrong parameters value: file %s on line %d\r\n", file, line) */
  /* USER CODE END 6 */
}
#endif /* USE_FULL_ASSERT */
